"""
加载字典模型
"""
import pickle
import torch


# 定义训练时的batch_size
train_batch_size = 128
# 定义测试时的batch_size
test_batch_size = 512

# 字典模型的加载
ws = pickle.load(open("./model/ws.pkl", "rb"))

# 句子的最大长度
seq_len = 200

# LSTM
# 输入数据的形状
input_size = 200
# 隐藏层LSTM单元数量
hidden_size = 128
# 隐藏层数量
num_layers = 2
# 设置dropout
dropout = 0.4

#
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

